from flask import g, request


# 自定义一个登录认证的装饰器,判断用户是否登陆了。如果登陆，继续访问。如果没有登陆直接返回（不能继续访问）。本质上就是一个登陆的拦截器
def login_required(func):
    def wrapper(*args, **kwargs):
        if g.user_id is not None:  # 表示已经登陆过
            return func(*args, **kwargs)
        else:
            return {'message': '用户没有登陆，不能继续访问'}, 401  # 注意在后面加状态码就是直接给出服务器的报错

    return wrapper
